package com.heima.media.controller;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.read.listener.ReadListener;
import com.heima.common.dto.ResponseResult;
import com.heima.media.entity.WmNews;
import com.heima.media.entity.WmSensitive;
import com.heima.media.service.IAuditService;
import com.heima.media.service.IWmNewsService;
import com.heima.media.service.IWmSensitiveService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.List;

@RestController
@RequestMapping("/test")
public class TestController {

    @Autowired
    private IWmNewsService newsService;

    @Autowired
    private IAuditService auditService;

    @GetMapping("/audit/{id}")
    public ResponseResult audit(@PathVariable("id") Integer id) {
        WmNews wmNews = newsService.getById(id);
        auditService.audit(wmNews);
        return ResponseResult.okResult();
    }
    /**
     * 文件上传
     * <p>1. 创建excel对应的实体对象 参照{@link WmSensitive}
     * <p>2. 由于默认一行行的读取excel，所以需要创建excel一行一行的回调监听器，参照{@link ReadListener}
     * <p>3. 直接读即可
     */
    @PostMapping("upload")
    @ResponseBody
    public String upload(MultipartFile file) throws IOException {
        EasyExcel.read(file.getInputStream(), WmSensitive.class, new ReadListener<WmSensitive>() {
            @Override
            public void invoke(WmSensitive o, AnalysisContext analysisContext) {
                // 每读到一行数据执行该方法
                System.out.println(o);
                // todo 将读取到的敏感词更新到数据库中
            }

            @Override
            public void doAfterAllAnalysed(AnalysisContext analysisContext) {
                // 所有数据读取完成执行
                System.out.println("所有数据读取完成");
            }
        }).sheet().doRead();
        return "success";
    }

    /**
     * 文件下载（失败了会返回一个有部分数据的Excel）
     * <p>
     * 1. 创建excel对应的实体对象 参照{@link WmSensitive}
     * <p>
     * 2. 设置返回的 参数
     * <p>
     * 3. 直接写，这里注意，finish的时候会自动关闭OutputStream,当然你外面再关闭流问题不大
     */
    @GetMapping("download")
    public void download(HttpServletResponse response) throws IOException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("敏感词", "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), WmSensitive.class).sheet("sheet").doWrite(data());
    }
    @Autowired
    private IWmSensitiveService sensitiveService;

    /**
     * 查询敏感词
     * @return
     */
    private List<WmSensitive> data() {
        return sensitiveService.list();
    }
}
